import torch

x  = torch.tensor([3,2,43,4])
y = torch.argmax(x,dim=0)
print(y)